package com.michale.system.service.impl;

import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.michale.common.constants.UserConstant;
import com.michale.common.domain.permissions.RoleDeptEntity;
import com.michale.common.domain.permissions.RoleEntity;
import com.michale.common.domain.permissions.RoleMenuEntity;
import com.michale.common.domain.permissions.UserRoleEntity;
import com.michale.common.exception.ServiceException;
import com.michale.common.utils.String.StringUtils;
import com.michale.common.utils.security.SecurityUtils;
import com.michale.system.mapper.permissions.RoleDeptMapper;
import com.michale.system.mapper.permissions.RoleMapper;
import com.michale.system.mapper.permissions.RoleMenuMapper;
import com.michale.system.mapper.permissions.UserRoleMapper;
import com.michale.system.service.RoleService;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;

import javax.annotation.Resource;
import java.util.*;

/**
 * @author Michale
 * @apiNote 角色表管理 业务层
 */
@Slf4j
@Service("roleService")
public class RoleServiceImpl extends ServiceImpl<RoleMapper, RoleEntity> implements RoleService {

    @Resource
    private RoleMenuMapper roleMenuMapper;
    @Resource
    private RoleDeptMapper roleDeptMapper;
    @Resource
    private UserRoleMapper userRoleMapper;

    /**
     * 根据条件查询角色数据
     *
     * @param roleEntity 角色信息
     * @return 角色数据集合信息
     */
    @Override
    public List<RoleEntity> selectRoleList(RoleEntity roleEntity) {
        return baseMapper.selectRoleList(roleEntity);
    }

    /**
     * 根据用户Id查找用户权限信息
     *
     * @param userId 用户ID
     * @return 用户权限信息
     */
    @Override
    public Set<String> selectPermissionByUserId(Long userId) {
        //通过用户ID查询角色信息
        List<RoleEntity> roleEntityList = baseMapper.selectRolePermissionByUserId(userId);
        //封装权限信息
        Set<String> permsSet = new HashSet<>();
        for (RoleEntity role : roleEntityList) {
            String roleKey = role.getRoleKey();
            permsSet.addAll(Arrays.asList(roleKey.trim().split(",")));
        }
        return permsSet;
    }

    /**
     * 根据用户ID获取角色选择框列表
     *
     * @param userId 用户ID
     * @return 选中角色ID列表
     */
    @Override
    public List<Long> selectRoleListByUserId(Long userId) {
        return baseMapper.selectRoleIdListByUserId(userId);
    }

    /**
     * 查询所有角色信息
     *
     * @return 角色信息列表
     */
    @Override
    public List<RoleEntity> selectAllRole() {
        return baseMapper.selectRoleList(new RoleEntity());
    }

    /**
     * 批量删除用户角色关联
     *
     * @param userIds 需要删除的角色ID
     * @return 结果
     */
    @Override
    public int deleteUserRole(Long[] userIds) {
        return baseMapper.deleteUserRoleByUserIds(userIds);
    }

    /**
     * 根据用户ID查询角色列表
     *
     * @param userId 用户ID
     * @return 角色列表
     */
    @Override
    public List<RoleEntity> selectRoleByUserId(Long userId) {
        return baseMapper.selectRoleListByUserId(userId);
    }

    /**
     * 校验角色名称是否唯一
     *
     * @param roleEntity 角色信息
     * @return 结果
     */
    @Override
    public String checkRoleNameUnique(RoleEntity roleEntity) {
        long roleId = StringUtils.isNull(roleEntity.getRoleId()) ? -1L : roleEntity.getRoleId();
        RoleEntity info = baseMapper.checkRoleNameUnique(roleEntity.getRoleName());
        if (StringUtils.isNotNull(info) && info.getRoleId() != roleId) {
            return UserConstant.NOT_UNIQUE;
        }
        return UserConstant.UNIQUE;
    }

    /**
     * 校验角色权限是否唯一
     *
     * @param roleEntity 角色信息
     * @return 结果
     */
    @Override
    public String checkRoleKeyUnique(RoleEntity roleEntity) {
        long roleId = StringUtils.isNull(roleEntity.getRoleId()) ? -1L : roleEntity.getRoleId();
        RoleEntity info = baseMapper.checkRoleKeyUnique(roleEntity.getRoleKey());
        if (StringUtils.isNotNull(info) && info.getRoleId() != roleId) {
            return UserConstant.NOT_UNIQUE;
        }
        return UserConstant.UNIQUE;

    }

    /**
     * 通过角色ID查询角色使用数量
     *
     * @param roleId 角色ID
     * @return 结果
     */
    @Override
    public int countUserRoleByRoleId(Long roleId) {
        return baseMapper.countUserRoleByRoleId(roleId);
    }

    /**
     * 新增角色信息
     *
     * @param roleEntity 角色对象
     * @return 结果
     */
    @Transactional(rollbackFor = {Exception.class})
    @Override
    public int insertRole(RoleEntity roleEntity) {
        roleEntity.setCreateBy(SecurityUtils.getUsername());
        baseMapper.insertRole(roleEntity);
        return insertRoleMenu(roleEntity);
    }


    /**
     * 修改角色状态
     *
     * @param roleEntity 角色信息
     * @return 结果
     */
    @Override
    public int updateRoleStatus(RoleEntity roleEntity) {
        roleEntity.setUpdateBy(SecurityUtils.getUsername());
        return baseMapper.updateRole(roleEntity);
    }

    /**
     * 批量删除角色信息
     *
     * @param roleIds 需要删除的角色ID
     * @return 结果
     */
    @Override
    public int deleteRoleByIds(Long[] roleIds) {
        // 删除角色与菜单关联
        roleMenuMapper.deleteRoleMenu(roleIds);
        // 删除角色与部门关联
        roleDeptMapper.deleteRoleDept(roleIds);
        return baseMapper.deleteRoleByIds(roleIds);
    }

    /**
     * 通过角色ID删除角色
     *
     * @param roleId 角色ID
     * @return 结果
     */
    @Override
    public int deleteRoleById(Long roleId) {
        return baseMapper.deleteRoleById(roleId);
    }

    /**
     * 取消授权用户角色
     *
     * @param userRoleEntity 用户和角色关联信息
     * @return 结果
     */
    @Override
    public int deleteAuthUser(UserRoleEntity userRoleEntity) {
        return userRoleMapper.deleteUserRoleInfo(userRoleEntity);
    }

    /**
     * 批量授权用户角色
     *
     * @param roleId  角色ID
     * @param userIds 需要删除的用户数据ID
     * @return 结果
     */
    @Override
    public int insertAuthUsers(Long roleId, Long[] userIds) {
        // 新增用户与角色管理
        List<UserRoleEntity> list = new ArrayList<>();
        for (Long userId : userIds) {
            UserRoleEntity ur = new UserRoleEntity();
            ur.setUserId(userId);
            ur.setRoleId(roleId);
            list.add(ur);
        }
        return baseMapper.batchUserRole(list);
    }

    /**
     * 批量取消授权用户角色
     *
     * @param roleId  角色ID
     * @param userIds 需要取消授权的用户数据ID
     * @return 结果
     */
    @Override
    public int deleteAuthUsers(Long roleId, Long[] userIds) {
        return baseMapper.deleteUserRoleInfos(roleId, userIds);
    }

    /**
     * 根据角色ID获取详细信息
     *
     * @param roleId 角色ID
     * @return 结果
     */
    @Override
    public RoleEntity selectRoleById(Long roleId) {
        return baseMapper.selectRoleById(roleId);
    }

    /**
     * 修改保存角色信息
     *
     * @param roleEntity 角色信息
     * @return 结果
     */
    @Override
    @Transactional
    public int updateRole(RoleEntity roleEntity) {
        // 修改角色信息
        roleEntity.setUpdateBy(SecurityUtils.getUsername());
        baseMapper.updateById(roleEntity);
        // 删除角色与菜单关联
        roleMenuMapper.deleteRoleMenuByRoleId(roleEntity.getRoleId());
        return insertRoleMenu(roleEntity);
    }


    /**
     * 根据用户ID查询角色
     *
     * @param userName 用户名
     * @return 角色列表
     */
    @Override
    public List<RoleEntity> selectRolesByUserName(String userName) {
        return baseMapper.selectRolesByUserName(userName);
    }

    /**
     * 新增角色和部门信息（数据权限）
     *
     * @param roleEntity 角色信息
     */
    private int insertRoleDept(RoleEntity roleEntity) {
        int rows = 1;
        // 新增角色与部门（数据权限）管理
        List<RoleDeptEntity> list = new ArrayList<>();
        for (Long deptId : roleEntity.getDeptIds()) {
            RoleDeptEntity rd = new RoleDeptEntity();
            rd.setRoleId(roleEntity.getRoleId());
            rd.setDeptId(deptId);
            list.add(rd);
        }
        if (list.size() > 0) {
            rows = baseMapper.batchRoleDept(list);
        }
        return rows;
    }

    /**
     * 新增角色和菜单关联信息
     *
     * @param roleEntity 角色对象
     * @return 结果
     */
    private int insertRoleMenu(RoleEntity roleEntity) {
        int rows = 1;
        // 新增用户与角色管理
        List<RoleMenuEntity> sysRoleMenuEntityList = new ArrayList<>();
        for (Long menuId : roleEntity.getMenuIds()) {
            RoleMenuEntity sysRoleMenuEntity = new RoleMenuEntity();
            sysRoleMenuEntity.setRoleId(roleEntity.getRoleId());
            sysRoleMenuEntity.setMenuId(menuId);
            sysRoleMenuEntityList.add(sysRoleMenuEntity);
        }
        if (sysRoleMenuEntityList.size() > 0) {
            rows = baseMapper.batchRoleMenu(sysRoleMenuEntityList);
        }
        return rows;
    }

    /**
     * 修改数据权限信息
     *
     * @param roleEntity 角色信息
     * @return 结果
     */
    @Override
    @Transactional
    public int authDataScope(RoleEntity roleEntity) {
        roleEntity.setUpdateBy(SecurityUtils.getUsername());
        // 修改角色信息
        baseMapper.updateById(roleEntity);
        // 删除角色与部门关联
        roleDeptMapper.deleteRoleDeptByRoleId(roleEntity.getRoleId());
        // 新增角色和部门信息（数据权限）
        return insertRoleDept(roleEntity);
    }

    /**
     * 判断登录者是否具有身份标识
     *
     * @param roleEntity   角色信息
     * @param roleClass 角色类型 (loginUser:登录者 operateUser:被操作的用户)
     */
    @Override
    public void checkRoleIdentity(RoleEntity roleEntity, String roleClass) {
        if (roleClass.equals(UserConstant.LOGIN_USER)) {
            if (StringUtils.isEmpty(SecurityUtils.getUserIdentity())) {
                throw new ServiceException("判断角色是否具有角色标识", "未为您分配角色, 请联系管理员");
            }
        } else if (roleClass.equals(UserConstant.OPERATE_USER)) {
            if (StringUtils.isEmpty(roleEntity.getIdentity()))
                throw new ServiceException("判断角色是否具有角色标识", "当前角色未分配角色标识");
        } else {
            throw new ServiceException("判断角色是否具有角色标识", "无法识别的用户类型");
        }
    }

    /**
     * 检查角色是否允许操作
     *
     * @param roleEntity 角色信息
     */
    @Override
    public void checkRoleAllowed(RoleEntity roleEntity) {
        if (!SecurityUtils.getUserIdentity().equals(UserConstant.IS_ADMIN)) {
            if (StringUtils.isNotEmpty(roleEntity.getIdentity())) {
                checkRoleIdentity(roleEntity, UserConstant.LOGIN_USER);
                checkRoleIdentity(roleEntity, UserConstant.OPERATE_USER);
                int i = SecurityUtils.getUserIdentity().compareTo(roleEntity.getIdentity());
                if (StringUtils.isEmpty(SecurityUtils.getUserIdentity())) {
                    throw new ServiceException("检查角色是否允许操作", "未为您分配角色, 请联系管理员");
                }
                if (i == 0) {
                    throw new ServiceException("检查角色是否允许操作", "不允许操作同级角色");
                }
                if (i > 0) {
                    throw new ServiceException("检查角色是否允许操作", "不允许操作上级角色");
                }
            }
            throw new ServiceException("检查角色是否允许操作", "当前角色未分配角色标识");
        }
    }

    /**
     * 校验角色是否有数据权限
     *
     * @param roleId 角色ID
     */
    @Override
    public void checkRoleDataScope(Long roleId) {
        if (!SecurityUtils.isAdmin(SecurityUtils.getUserIdentity())) {
            RoleEntity roleEntity = new RoleEntity();
            roleEntity.setRoleId(roleId);
            List<RoleEntity> roleEntityList = baseMapper.selectRoleList(roleEntity);
            if (StringUtils.isEmpty(roleEntityList)) {
                throw new ServiceException("没有权限访问角色数据！");
            }
        }
    }
}
